from env_tools import get_env_value
from langchain_deepseek import ChatDeepSeek
from langchain_core.messages import SystemMessage, HumanMessage

# 获取apikey
api_key = get_env_value("DEEPSEEK_API_KEY")
if api_key is None:
    raise ValueError("请设置环境变量DEEPSEEK_API_KEY")

# 初始化deepseek
model = ChatDeepSeek(
    model = "deepseek-chat",            # 模型名称
    temperature = 0,                    # 采样温度，指定0表示输出更确定性，取值在[0,1]之间，越小越精确，通常设置为0.7
    max_tokens=None,                    # 输出最大长度，None表示无限长度
    timeout=None,                       # 请求超时，None表示无限超时
    max_retries=2,                      # 最大重试次数
    api_key=api_key
)

# 创建消息列表
messages= [
    SystemMessage(content="你是一个充满无限慧的智能力，你的名字叫天鉴"), # 系统提示词
    HumanMessage(content="你是谁？")    # 用户输入
]
# 调用大模型
resp = model.invoke(messages)
# 打印输出
print(resp.content)
# print(resp.usage)
print(resp.type)
print(resp.usage_metadata)
print(type(resp))
